"""
需要的浏览器配置对象
"""
import datetime
import os
import sys

from selenium import webdriver
from selenium.webdriver.edge.service import Service
from webdriver_manager.microsoft import EdgeChromiumDriverManager


class webDriver:
    driver = ""

    def __init__(self):
        options = webdriver.EdgeOptions()
        # 无头模式
        # options.add_argument("--headless")
        # 窗口最大化
        # options.add_argument("--start-maximized")
        # 添加页面加载选项
        # options.page_load_strategy = "eager"
        self.driver = webdriver.Edge(service=Service(().install()), options=options)
        # 添加隐式等待2秒
        self.driver.implicitly_wait(2)

    # 添加屏幕截图
    def GetScreenShot(self):
        # 使用日期时间命名文件夹
        dirname = datetime.datetime.now().strftime("%Y-%m-%d")
        # 如果路径不存在就重新创建文件夹
        if not os.path.exists("../images/" + dirname):
                os.mkdir("../images/" + dirname)
        # 方法名+时间
        filename = sys._getframe().f_back.f_code.co_name + "-" + datetime.datetime.now().strftime("%Y-%m-%d-%H%M%S")     + ".png"
        # 文件夹+文件夹名+文件名
        self.driver.save_screenshot("../images/" + dirname + "/" + filename)


ForumDriver = webDriver()
# login = webDriver()
# login
